import os  
import pandas as pd  
from glob import glob  
  
# 设置文件夹路径  
folder_path = 'C:\Big X\ECUST\数学与应用数学\毕业设计\机械结构\磨床主轴\CloudPoint'  # 替换为你的文件夹路径  
  
# 获取文件夹中所有的.txt文件  
txt_files = glob(os.path.join(folder_path, '*.txt'))  
  
# 循环遍历每个文件  
for index, file_path in enumerate(txt_files, start=1):  
    # 读取.txt文件  
    df = pd.read_csv(file_path, sep='\s+', header=None)  # 假设文件使用空格分隔，没有标题行  
      
    # 替换r, g, b列为循环次数  
    df.iloc[:, 3] = index  # 假设r是第四列（索引为3）  
    df.iloc[:, 4] = index  # 假设g是第五列（索引为4）  
    df.iloc[:, 5] = index  # 假设b是第六列（索引为5）  
      
    # 构建新的文件名（可选）  
    new_file_path = f"{file_path[:-4]}_modified.txt"  # 在原文件名后添加_modified后缀  
      
    # 将修改后的数据保存回新的.txt文件  
    df.to_csv(new_file_path, sep=',', header=False, index=False)  # 使用空格分隔，不保存标题行和索引  
  
print("处理完成！")